package com.itheima.reggie.service.impl;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.reggie.common.CustomException;
import com.itheima.reggie.entity.Category;
import com.itheima.reggie.entity.Dish;
import com.itheima.reggie.entity.Setmeal;
import com.itheima.reggie.mapper.CategoryMapper;
import com.itheima.reggie.service.CategoryService;
import com.itheima.reggie.service.DishService;
import com.itheima.reggie.service.SetmealService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

/**
 * 分类管理的实现方法
 */
@Service
public class CategoryServiceImpl extends ServiceImpl<CategoryMapper, Category> implements CategoryService {
    @Autowired
    private DishService dishService;

    @Autowired
    private SetmealService setmealService;

    /**
     * 根据 id 删除分类，删除之前需要进行关联判断
     * @param ids 要删除的分类id
     */
    @Override
    public void remove(Long ids) {
        // 查看当前分配是否关联了菜品，如果已经关联，则无法进行删除
        LambdaQueryWrapper<Dish> dishWrapper = new LambdaQueryWrapper<>();
        // 指定要查询的 id
        dishWrapper.eq(Dish::getCategoryId, ids);
        // 调用 dish 的 mapper 进行查询与当前分类相关的菜品条数
        int dishCount = dishService.count(dishWrapper);
        // 如果查询得到的关联数量不为0，说明已经有菜品属于当前分类，抛出异常
        if (dishCount != 0) {
            throw new CustomException("当前分类下关联了菜品，不能删除");
        }

        // 查看当前分配是否关联了套餐，如果已经关联，则无法进行删除
        LambdaQueryWrapper<Setmeal> setmealWrapper = new LambdaQueryWrapper<>();
        // 指定要查询的 id
        setmealWrapper.eq(Setmeal::getCategoryId, ids);
        // 调用 setmeal 的 mapper 进行查询与当前分类相关的套餐条数
        int setmealCount = setmealService.count(setmealWrapper);
        // 如果查询得到的关联数量不为0，说明已经有菜品属于当前套餐，抛出异常
        if (setmealCount != 0) {
            throw new CustomException("当前分类下关联了套餐，不能删除");
        }

        // 如果没有任何关联，则正常删除当前分类
        super.removeById(ids);
    }
}
